package controller;

import org.apache.commons.io.IOUtils;
import sun.misc.BASE64Encoder;

import javax.servlet.ServletContext;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;

/**
 * @author HLF
 * @version V1.0
 * @package controller
 * @date 2021/11/3 10:12
 */
public class DownLoadServlet extends HttpServlet {
    @Override
    protected void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {
        //1.获取要下载的文件名
        String downloadFileName = "1001.jpg";
        //2.读取要下载的文件内容（通过ServletContext对象可以获取）
        ServletContext servletContext = getServletContext();
        //获取要下载的文件类型（找到指定路径下的文件）
        //http://localhost:8080/imgs/userUpload/文件名
        String mimeType = servletContext.getMimeType("/imgs/userUpload/" + downloadFileName);
        System.out.println("要下载的文件类型是:" + mimeType);

        //3.在回传前通过响应头告诉客户端返回的数据类型
        resp.setContentType(mimeType);

        //4.还要告诉客户端收到的数据是用于下载使用(通过响应头)
        // Content-Disposition 响应头，表示收到的数据怎么处理
        // attachment 表示附件，表示下载使用
        // filename= 表示指定下载的文件名

        //resp.setHeader("Content-Disposition","attachment;filename="+downloadFileName);
        //url编码是把汉字转换成%xx%xx的格式
        //resp.setHeader("Content-Disposition","attachment;filename="+ URLEncoder.encode("javaee三层架构.jpg","UTF-8"));//可以和原文件名字不一样

        //String str = "attachment;filename=" + URLEncoder.encode("admin.jpg", "UTF-8");
        //resp.setHeader("Content-Disposition", str);

        String ua = req.getHeader("User-Agent"); // 判断是否是火狐浏览器
        String str = null;
        if (ua.contains("Firefox")) {//如果是火狐浏览器
            System.out.println("火狐浏览器");
            // 使用下面的格式进行 BASE64 编码后
            str = "attachment; fileName=" + "=?utf-8?B?"
                    + new BASE64Encoder().encode("头像.jpg".getBytes("utf-8")) + "?=";
            // 设置到响应头中
        } else {//如果是其他浏览器
            System.out.println("其他浏览器");
            // 把中文名进行 UTF-8 编码操作。
            str = "attachment; fileName=" + URLEncoder.encode("头像.jpg", "UTF-8");
            // 然后把编码后的字符串设置到响应头中
        }
        resp.setHeader("Content-Disposition", str);

        //5.把下载的内容回传给客户端
        //通过输入流读取要下载的内容
        InputStream resourceAsStream = servletContext.getResourceAsStream("/imgs/userUpload/" + downloadFileName);
        //获取响应的输出流
        OutputStream outputStream = resp.getOutputStream();
        //读取输入流中全部的数据，复制给输出流，输出给客户端
        IOUtils.copy(resourceAsStream, outputStream);
    }
}
